'''------------------------------------------------------------------------------------------------
 $ @Author: your name
 $ @Date: 2021-12-23 19:23:21
 $ @LastEditTime: 2021-12-24 21:52:40
 $ @LastEditors: Please set LastEditors
 $ @Description: 打开koroFileHeader查看配置 进行设置: https://github.com/OBKoro1/koro1FileHeader/wiki/%E9%85%8D%E7%BD%AE
 $ @FilePath: \YSQTrackerEvaluationToolKit_v1.0\GetCfg\setParam_OutPutVisRes.py
--------------------------------------------------------------------------------------------------'''
import argparse

def getParam():
    param = argparse.ArgumentParser("add param")
    param.add_argument(
        '-fps',
        '--fps',
        default=25,
        type = int,
        help = '输出可视化结果为视频文件时，指定的视频文件帧率'
    )
    #* 可视化结果输出形式-> imgFile 为图像集合； video 为视频文件; N 不输出;B两个格式都输出
    param.add_argument(
        '-ovrf',
        '--OutputVisualizationResultsFormat',
        default='B',
        help = '可视化结果输出形式-imgFile,video,N'
    )

    param.add_argument(
            '-vnld',
            '--videosNameListDir',
            default=r'.\GETCFG\videosNameList_OutPutVisRes.yaml',
            type = str,
            help = " "
        )
    param.add_argument(
            '-tsd',
            '--testSetDir',
            default=r'F:\tracking_test_DataSet\OTB100_v0',
            type = str,
            help = "OTB数据集的root dir"
        )
    param.add_argument(
            '-tbd',
            '--trackerBboxDir',
            default={
                'C-COT':r'.\tarckers\CCOT',
                'DaSiamRPN': r'.\tarckers\DaSiamRPN',
            #         'ECO':r'D:\PY_HC\YSQTrackerEvaluationToolkit\tarckers\ECO',
            #         'ECO-HC':r'D:\PY_HC\YSQTrackerEvaluationToolkit\tarckers\ECO-HC',
            #         'MDNet':r'D:\PY_HC\YSQTrackerEvaluationToolkit\tarckers\MDNet',
            #         'Ocean':r'D:\PY_HC\YSQTrackerEvaluationToolkit\tarckers\Ocean',
            #         'RTMDNet':r'D:\PY_HC\YSQTrackerEvaluationToolkit\tarckers\RTMDNet',
            #         'SiamBAN':r'D:\PY_HC\YSQTrackerEvaluationToolkit\tarckers\SiamBAN',
            #         'siamRCNN':r'D:\PY_HC\YSQTrackerEvaluationToolkit\tarckers\siamRCNN',
            #         'SiamRPN':r'D:\PY_HC\YSQTrackerEvaluationToolkit\tarckers\SiamRPN',
            #         'SiamRPN++':r'D:\PY_HC\YSQTrackerEvaluationToolkit\tarckers\SiamRPN++',
            #         'SiamFC-3S':r'D:\PY_HC\YSQTrackerEvaluationToolkit\tarckers\SiamFC-3S',
            #         'UPDT':r'D:\PY_HC\YSQTrackerEvaluationToolkit\tarckers\UPDT',
                    # 'siamfc':r'D:\Experimaental_Data\P_SiamFc\siamfc_0.5_biaozhun',
                    # 'our': r'D:\Experimaental_Data\P_SiamFc\2021_12_1_addEmbedding'
                    # 'OnlineUpdate':r'C:\Users\Administrator\Desktop\litongANs\txt—updata',
                    # 'NoUpudate':r'C:\Users\Administrator\Desktop\litongANs\txt'
            },
            help = "这是一个字典，key为跟踪器的名字，key对应的str为该跟踪器prebox存放的地址"
        )
    param.add_argument(
        '-sfef',
        '--starFrameAndEndFrame',
        default={
            'David': [300,0],
            'Football1':[0,73],
            'Human4':[0,666]
        }
    )
    return param.parse_args()